import csv

one_hot = [0 for i in range(6)]
one_hot[3] = 1
a = one_hot.index(max(one_hot))
print(a)
print(one_hot)
# print(rows)